import pymssql


class MSSQL:
    """
    这是对MSSQL数据的调用。
    """
    def __init__(self, server=None, user=None, password=None, database=None):
        if not all([server, user, password, database]):
            assert 'specify server, user, password, database parameter.'

        conn = pymssql.connect(server=server, user=user, password=password, database=database)
        self.cursor = conn.cursor()

    def query(self, sql: str) -> list:
        """
        这是一个通用的查询方法。
        """
        results = []
        self.cursor.execute(sql)
        while True:
            row = self.cursor.fetchone()
            if not row:
                break
            results.append(row)

        return results


